package com.lzl.logdesenshelper.utils;

import com.lzl.logdesenshelper.annotation.LogField;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;

/**
 * 脱敏工具类
 *
 * @author zrj
 * @since 2021/8/16
 **/
@Slf4j
public class DesensitizationUtil {


    public static String desentizeParameter(Object obj, LogField logField) {
        // 是否是接口
        if (obj.getClass().isInterface()) {
            return obj.toString();
        }
        if (TypeUtil.isPrimitive(obj.getClass())) {
            if (!(obj instanceof String)) {
                return obj.toString();
            }
            if (logField == null) {
                return obj.toString();
            } else {
                String param = obj.toString();
                return DesensitizationUtil.desensitizeValue(logField, param);
            }
        }
        return obj.toString();
    }



    public static String desensitizeValue(LogField logField, String valueStr) {
        switch (logField.value()) {
            case CHINESE_NAME: {
                return DesensitizationUtil.chineseName(valueStr);
            }
            case ID_CARD: {
                return DesensitizationUtil.idCardNum(valueStr);
            }

            case MOBILE: {
                return DesensitizationUtil.mobilePhone(valueStr);
            }
            case ADDRESS: {
                return DesensitizationUtil.address(valueStr, 4);
            }
            case EMAIL: {
                return DesensitizationUtil.email(valueStr);
            }
            case BANK_CARD: {
                return DesensitizationUtil.bankCard(valueStr);
            }
            case IGNORE: {
                return null;
            }
            default: {
                break;
            }
        }
        return null;
    }



    /**
     * [中文姓名] 只显示第一个汉字，其他隐藏为2个星号<例子：李**>
     *
     * @param fullName 中文姓名
     * @return 中文姓名
     */
    private static String chineseName(String fullName) {
        if (StringUtils.isBlank(fullName)) {
            return "";
        }
        String name = StringUtils.left(fullName, 1);
        return StringUtils.rightPad(name, StringUtils.length(fullName), "*");
    }


    /**
     * [身份证号] 显示最后四位，其他隐藏。共计18位或者15位。<例子：*************5762>
     *
     * @param id 身份证号
     * @return 身份证号
     */
    private static String idCardNum(String id) {
        if (StringUtils.isBlank(id)) {
            return "";
        }
        String num = StringUtils.right(id, 4);
        return StringUtils.leftPad(num, StringUtils.length(id), "*");
    }


    /**
     * [手机号码] 前三位，后四位，其他隐藏<例子:138******1234>
     *
     * @param num 手机号码
     * @return String
     */
    private static String mobilePhone(String num) {
        if (StringUtils.isBlank(num)) {
            return "";
        }
        return StringUtils.left(num, 3).concat(StringUtils.removeStart(StringUtils.leftPad(StringUtils.right(num, 4), StringUtils.length(num), "*"), "***"));
    }

    /**
     * [地址] 只显示到地区，不显示详细地址；我们要对个人信息增强保护<例子：北京市海淀区****>
     *
     * @param address       地址
     * @param sensitiveSize 敏感信息长度
     * @return String
     */
    private static String address(String address, int sensitiveSize) {
        if (StringUtils.isBlank(address)) {
            return "";
        }
        int length = StringUtils.length(address);
        return StringUtils.rightPad(StringUtils.left(address, length - sensitiveSize), length, "*");
    }

    /**
     * [电子邮箱] 邮箱前缀仅显示第一个字母，前缀其他隐藏，用星号代替，@及后面的地址显示<例子:g**@163.com>
     *
     * @param email 电子邮箱
     * @return String
     */
    private static String email(String email) {
        if (StringUtils.isBlank(email)) {
            return "";
        }
        int index = StringUtils.indexOf(email, "@");
        if (index <= 1) {
            return email;
        } else {
            return StringUtils.rightPad(StringUtils.left(email, 1), index, "*").concat(StringUtils.mid(email, index, StringUtils.length(email)));
        }
    }

    /**
     * [银行卡号] 前六位，后四位，其他用星号隐藏每位1个星号<例子:6222600**********1234>
     *
     * @param cardNum 银行卡号
     * @return String
     */
    private static String bankCard(String cardNum) {
        if (StringUtils.isBlank(cardNum)) {
            return "";
        }
        return StringUtils.left(cardNum, 6).concat(StringUtils.removeStart(StringUtils.leftPad(StringUtils.right(cardNum, 4), StringUtils.length(cardNum), "*"), "******"));
    }

}

